Fixed wk1 ex6 to return a value