Cleaning up imports for console.py