Add the solution as well as some code for the database connection