Correcting some typos on the code