Python Returning An Array Element