Python script that parses a csv file