Updated comments

Comments updated to reflect changes