Add a simple state machine to BigBird