Refactored the internal wave code. Computing all fields now, but there is clearly a normalization issue that I need to take care of.