Fixed typo and added some more comments