Added test and logic for no return branches.