Adding extension class