Import get_user_model or define it