Beautify example, removes unit_square var.