[ADD] Add find_model function