Slight code formatting