added a return value and started challenge 2