Better formatting

Forgot to add some formatting changes