Add return statements to CFG