Update alert dialog code