Updated comments

Added thanks to contributer