Update simple_boxes.py

cleaned up