Trying new algorithms

Cleaned up the code and used different classifier. Also implemented a
better way to fill in missing data from training set.