Refactor: added class documentation