train is done, need to check