Add step piece movement logic