Remove unnecessary imports
Add setup.py to support easy installation
Rename project to viridis