HW2 attempt 2