Added FSM for loading