Update staircases.py