HW3 working now