Wave equation

**ToDo**

- [ ] Make rewrite methods work