Updated formatting of code