Hook up the onboarding code to work with the new state system