minor changes

minor changes and added comments