CLeaned up the styleing of the test