Added extension test code