Added TestUI class