Updated code to handle with pycapsules. Assignment of memory is only done within the multiprocessing process/thread, otherwise we cant  use multiprocessing