Update to use pg package