Added ability to simulate energy