Fixes, code rearrangement.