Create energy.py