Implemented conditional branching in TestStep trait