train implemented but not done yet, going to eat