Fixed bug with csv logger
Removed unnecessary print statement
Made prediction step a variable, and set it to 10