added comments to collect energy python file. this is also in model analysis now, probably should use that version