Update dialog code