Week 2 save function working