Updated some more methods in WormTest