Added the update functionality, modified the UI