Make code Python 2 and 3 compatible

Here you go :)