Update for python 3
Merge pull request #5 from jayvius/python3

Update for python 3