Add dropout, softmax layer.