Match class edited
make_prediction removed puts statements
match tests edited