Added Branch support to code generation