Can now predict and learn from trainign set