Add branch coverage