Add Riemann solver and tests.
Fix code for Python 3.
Refactor: rename u to v.