Update curses_test.py

Cleaned up a little