small changes and comments added