finished HW with class help