Update 2

Updated comments