Impleement a more complex integration test
Separate display responsibilities from main script