Improve vim setup script