Hopefully completed the save in session part of the assignment