Added database connection and test to see if working correctly