Add state to determine what the loading flow is doing