Database setup working, added unit test for database