Made python 2 compatible

Adding from __future__ import print_function for python 2.x compatibility