Pylint fixes

Add print_function import for Python2
Whitespace fix