Some small changes to tyson_oscillator.py