Add path argument to the load_heap() function