Adding boundary visualization