Switch to using main states and switch to using localization for position