tidying up code formatting a bit.