Actually lets afterPenalty run properly. Adds a flag for debugging in PenaltyStates. Closes #534