Add a few tests