Initial working code, with test.py.