Homework 1 for CodePath Swift class