Updated comments on code