Add update functionality.
Update tests & README