Added the fuction tidy_up to preprocess the data before training. The
function clean doesn't make sense anymore.