Updated comments

Updated the comments so we can implement the changes