How to use sort() in Python