Adding functionality 2 to test.code