Update square basin example