Some final tweaks on the internal wave example code.