Add findclass function