Update wave equation implementation