Some changes to python setup