Need gradient check + Tweaking params