Add save method test