Add arnav's tests