Add test script
Fix bugs