Edited p_wrapper.py