Added main class for the mod