updated setup to support db