Changed extension class in main.