Adds `setup_database` to pass an already setup database connection to use
instead of `connect`