Create prism.py