use np to save output