Add stylistic changes to the code