Just have to add the update query now