Use self.class instead of hardcoding the class name