Added some comments. Minor changes