Updated comments in the code