Another fix for stairs program