Added missing method in class