Fixed comments on code