Ensure that ones_like is used for numpy 1.6; also add _arg