Added comment on class