Add state transitions to UI