Update Katka.py

just some clean up