add training mode