GRADIENT CHECK WORKS FOR W!!!!!!