Add: various improvements and fixings to code section support.