Adding TODO for class2