added more to save function