Update ShapeLogic.py