Add test mode