Remove TestClass and add refresh button