updating tests further