Added a test for the database