Add vim modelines to python files