Add python script to parse the output