Iron out your python code