Readme and small code cleanup
Working on training data