Additional prep for quick multiball

Changed ball stage check to be usable from other modes, not just from
the base mode at the start of a ball.