Fit some driver bug and add a simple demo `spectrum.py`