Added in python 2 support using import __future__