Added logic to create the save dir if it doesn't exist.