Properly flesh out example
Tweak test again