modified training and added code to clean data