Create diamond.py