Add save code feature