Add localization stage