Made fixes to Python pipeline