Update gradient-checking.jl