Modified to use Postgres in production