added test for testing how well model learned