Added the gradient check. Also removed the unnecessary function for computing the difference between gold/prediction gradients.