Better repo layout, fixed bug in functions for reading data

Better repo layout and fixed bug in functions for reading training data