Basic bubble sort in python