Answer to energy diagram problem