Updated the UI because of the new state machine.