Fixd issue in ex4 end solution