Added code to second branch