Simple verison of train classifier