vim: make the Python script snippet faster