Add `before` and `after` to track metrics at each step

[slinlee]

to help with writing reward terms in the webapp, let's track the metric values before and after that step.