Update test.py

Curses pre-work. Toying with the UI, etc.