Removed non-KNN statements from the main script