update demo & remove test code