Change updateWithMedia to use new method, add test