Clean up _heap.py to have a single function `initialise` for flag, u and heap.