Data.py is now great, working on fixing KNN