Add initial test code