Some cosmetic changes

Changed decision tree for py2 / py3 to simpler one, and changed the regex to work simpler.

Please don't bump the version, I'll do that once i push it to pypi.