Next step: add update function