Modify method ex4 to pass test > green