add changes to update_account method