Minor changes.

Signed-off-by: deepak <dpak.cherian@gmail.com>
minor bugfix.

Signed-off-by: deepak <dpak.cherian@gmail.com>
Gradient checking code.

Signed-off-by: deepak <dpak.cherian@gmail.com>