Modified Main test class, deleted scala main test class.