Ex#4
working on the return