Add vim wrapper class with tests
Refactor UI class to not call vim module directly