Implemented training and generating model <Karthik>
WIP - Using generated model to work on test data <Karthik>