Working code for Test A