Minor changes

Added some comments