Code comments updated