Added update functionality
(Almost) added update functionality