Work from DSP course in Python