Update stair.py