Added some UI test code