Add a displacement visualization toggle