Minor edit to riccati.py