Make pool class confirgurable and update simple test script