fixed test and train scripts so everything seems to work! 95.89% accuracy