Add some comments on class