Small Code changes

Added some comments