Fixing typo and alignment of code