updated TestMainClass