Reintroudce the test code to program