Some formatting in test file. Closing #2