Update setup.py

Add templates into the setup.