Wrap methods into class