Update to use pg