[Update] Add comments for classes