Changed to Maven project. Moved machine learning part out.