Fixed bug with changes updating. Added testing around updating the UI.