Minor fixes
Adding documentation comment