Week2 Save function working