Updated formatting of code

updated