first homework (second attempt)