Add tests for arena