Add method on_move