Add dropout for regularization.