Numpy Exercise – Solution