trying to to gradient checking. Not working.