Add Battle restriction