Create lorentz.py