added support for extra methods in generated class