Updated setup python file