Add dropout rate modification