GUI improvements, especially on the energy storage unit