Add test button and code