Fix stupid id mistake
Add update function