Add restrict optimization