Add coding in setup.py