Fix code whitespace/formatting