Python energy and forces also working