iOS-tut Assn4: simplify code