Trying to add save function
Adding more functions