Added Branch logic