Fixed typo on my last commit and prettify the code