Add db setup code