[python] Python: finding an element in a list