Fixed up spacing/prettyfying the code