added main class to __init__