Minor code formatting clean up