Switched to minitest
Use one loop to train clasiffier