HW2 attempt 1