Update StateManager.js