Python script that adds line numbers to a text file