successful gradient check for U, working on W now