Create einstein.py