Minor formatting to code