Changes to allow it to connect to the database