Small Update
Added some comments