updated code formatting

to be consistent with original style