Python Array Sort Function