Shortened Tower.py logic