Modify method ex5 to pass test > red