Fixing gradient computationgit push origin master