Add support for using class methods