adding some stuff to one of the Test classes