Minor formatting changes

Changed around the format of some code just a little