Making better style for cat.py and cat2.py