Time for action – adding the Awake method to StateManager