Make the profile_output script usable for Python 2 as well

scripts/profile_output.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3.6
+# -*- coding: utf-8 -*-
 """
 Profile a piece of Python code with ``profile``. Tries a completion on a
 certain piece of code.
@@ -18,7 +19,11 @@ Options:
 """
 
 import time
-import profile
+try:
+    # For Python 2
+    import cProfile as profile
+except ImportError:
+    import profile
 import pstats
 
 from docopt import docopt