wrote run_test and load_data methods, refactored some code