Almost there, just need to handle saving the data