refactor write stub in the tests
Add simple test on expected screen output
Add test for choices printed to screen