fixed two minor errors (missing colon and specified arg)
Merged with Matt's changes to assign rewards to nns at end of episode.
Trying to merge conflicts with Matt's changes on assigning rewards.