Add more code to support saved state