work from http://www.thinkful.com/courses/IOS-002/unit/3/lesson/4/assignment/1