Small update to tyson_oscillator.py