Added a function to find team