Updated some code to work with python3