Make it python3 compatible