minor changes

changed the CFG a bit, and added some comments