Implement find class function