Add a few happy path tests