Update geometry code