Updated line 142 for $user_class