Improve the example to show branch coverage clearly