wheeeeeeee

This works, but with my dataset it classifies everything as "beated"
I think there needs to be significantly more training data than test data, continuing to play with it..