Added in update functionality