Enhancing TWave

**ToDo**

- [x] Make rewrite methods work
I think it's ok to just return the lhs, since it's equated to 0. Otherwise you can use `Eq(lhs, 0)`.