updated for adding tests to pg