Renamed training set to test and corrected