Runs now with python 3 instead of python 2
Adapted for python3. Same output as I got with python2
Ignore .idea directory