Add tests

Should be easy to add tests with a sqlite memory db.