Cleanup and more JIT work. Checkpointing.