Added __future__ imports for Python 2 compatibility