updated test classs