Add energy visualisation demo