Imported print function for Python 2 compatibility
made tests pass