Add shape.py