Python Returning An Array From A Function