Add database creation code