with working gradient check for W,U,b...still need L