Add Functional Test Code