Clean up wave equation