Add code to update database schema