fixed setup.py and added main function