add check_energy.py