Tied the editing functionality together

This should be all the functionality for the demo.