Added toggle button for custom energy