Clean up the code. Now we will be able to use both the bruteforce + the nearest neighbor strategy