changed DialogFragment import and added a return