Make it python 3 compatible