Added a test DB
Made some changes to the models