exercise2 with vim and python