Modified code for user profile edit section