added comments
minor changes