text: Title: Write tests for __main__.py
  Body: As we can see on CodeClimate https://codeclimate.com/github/python-security/pyt/coverage/5935971dbf92ed000102998b there is pretty low test coverage of main, I understand why this is but adding some tests for it would increase our test coverage percentage and 75% isn't satisfying.
  
  If you have any trouble with this I can help, I am going to label this issue as Easy so new comers see it.