adding dropout flag