Comments in features
Minor code cleanup
Added code to train a model start to finish