Heal class and it's use in main.py