Refactor tests to Arrange, Act, Assert (AAA) style