added some ui test code