Modified main.py - add two classes for Karta and Talia