Updated arrange-act-assert headers for consistency