Got edit and save buttons to work