Format code

Just some clean up in order to be consistent with your CS.
I like to align them, but I won't mind if we do not :)