Update and rename if_train to Move to