Add code to clean up and remove excess tokens from python code.