correction and adding comments