Added code for handling the connection and creation of the database, however Test might not be correct