Updated setup for python 3