Updated comments

Added and removed comments