minor comments added
some comments and basic plotting