Updated tests

PR on your PR: the tests needed to be updated to match your formatting changes