Update to work with Sketch 3