test: using capsys
Removed test class