Comment by Narishma

Didn't you just shift the branch to the calculateDiscount() function?