HW5 as far as I could get for now