editing a text file in python