Adding comment and docs