You could put the EnablePlayerInput Function behind the InProgress Session State instead of in the CountDown Session State.
But everything is and should be happening within the CountDown Session State, I placed everything there.
Also Enabling Input could be done exactly like the disabling, except instead of setting FireButtonPressed, you just Get FireButtonPressed.