Correct walking logic