Add branch straigtening