add energy in one hard-coded potential
various train edits