adding dropout support