Fixed some formatting.

Adapted whitespace and indentation to fit in with the original code.