Changed display units to ns, cleaned up energy balance loop.