Updated comments and fixed typo