Fixed doc

Updated doc block comments.