Format code

Just some clean up in order to be consistent with your CS.