Apple Interview Question

class modification or adding some classes for existing code base