Fixed main() signature to work with pypi installer