Add test and attack fns (not working)