Add back the __init__ for UI base class

#638 was for taking advantage of inheritance for UI tests, but the `__init__` method was removed by mistake from base class. Adding it back again to really take advantage of inheritance for UI.