Updated example, added class documentation