Added grad&drop step