Move Python code into after/ftplugin file