added comments
minor fix