small fix and comments added