Removing the save to disk as it was not really necessary