Added plotting module and energy analysis