Modify design in README
Modify viralize.py