Modify edit user function