Added setup for db