Python Return An Array