second homework first task in swift