Fixed code formatting to make it more consistent