Add timeout tests