Add some happy path tests