HW2 Fixes