Documentation added

minor code fix