When I run the code on a bigger amount of data, it either gives an error or the results are not so adequate. Any idea what I can do to make it work?