Added non-linear boundary visualisation.