Updates to properly connect to the database