Working version, including image lasso, and correct caustics