Formatting of code + typo fixed