First pass at energy plotting is working