Cleaned up code style.

Follows the convention used in the majority of the codebase.