Added some documentation the class