Added more comments/clean-up.
Trainig finds good initial feature. Needs cleanup.