Fixed doctest for rollingdisc_example_lagrange

Fixed the doctest to pass after this change. Fixed code quality
issue in lagrange.py.