Bug fixes for multilayer plotting

Includes update to plotting of multilayer plane-wave example.