Added save code for numpy