Can you fix the following code:

$[$from qiskit import QuantumRegister, QuantumCircuit

from numpy import pi

def mark$\_$pure$\_$states$($qc$,$ b$0,$ b$1,$ b$2,$ b$3)$:

# mark the components corresponding to the pure states $|0011>,|1100>,|0101>$

# your code here

raise NotImplementedError

b $=$ QuantumRegister$(4,\text{'}$b$\text{'})$

qc $=$ QuantumCircuit$($b$)$

mark$\_$pure$\_$states$($qc$,$ b$[0],$ b$[1],$ b$[2],$ b$[3])$

qc$.$draw$(\text{'}$mpl$\text{'},$ style$=$'iqp'$)]$

such that it passes the following two tests.

Test $1$:

$[$from qiskit import QuantumCircuit, QuantumRegister, Aer, execute

b $=$ QuantumRegister$(4,\text{'}$b$\text{'})$

qc $=$ QuantumCircuit$($b$)$

# create uniform super position

qc$.$h$($b$[0])$

qc$.$h$($b$[1])$

qc$.$h$($b$[2])$

qc$.$h$($b$[3])$

qc$.$barrier$()$

mark$\_$pure$\_$states$($qc$,$ b$[0],$ b$[1],$ b$[2],$ b$[3])$

display$($qc$.$draw$(\text{'}$mpl$\text{'},$ style$=$'iqp'$))$

# use a state vector simulator to obain the marked states

backend $=$ Aer.get$\_$backend$(\text{'}$statevector$\_$simulator'$)$

job $=$ execute$($qc$,$ backend$)$

result $=$ job.result$()$

state$\_$vector $=$ result.get$\_$statevector$()$

print$($state$\_$vector$)$

for i in range$(16)$:

if i $==3$ or i $==10$ or i $==12$: # are we marking the correct basis states

assert abs$($state$\_$vector$[$i$]+0\mathrm{.}25)<=0\mathrm{.}001$

else:

assert abs$($state$\_$vector$[$i$]-0\mathrm{.}25)<=0\mathrm{.}001]$

Test $2$:

$[$# Let's run Grover's algorithm

# let's now implement Grover's search to find a value

import numpy as np

from qiskit import QuantumCircuit, QuantumRegister, Aer, execute

def Uf$($qc$,$ b$)$:

mark$\_$pure$\_$states$($qc$,$ b$[0],$ b$[1],$ b$[2],$ b$[3])$

def apply$\_$reflection$\_$about$\_$uniform$\_$state$($qc$,$ input$\_$registers$)$:

# qc is a previously created quantum circuit

# input$\_$registers are the registers that measure the input

# n is the number of input qubits

#

# $1.$ Apply Hadamard on each of the input registers.

# $2.$ Invert all the qubits

for i in input$\_$registers:

qc$.$h$($i$)$

qc$.$x$($i$)$

n $=$ len$($input$\_$registers$)$

# $3.$ apply a multi controlled Z gate

qc$.$mcp$($np$.$pi$,$ input$\_$registers$[0$:n$-1],$ input$\_$registers$[$n$-1])$

for i in input$\_$registers:

qc$.$x$($i$)$ # invert back

qc$.$h$($i$)$ # apply Hadamard back

def Grover$\_$diffuse$($qc$,$ inputs$)$:

Uf$($qc$,$ inputs$)$

apply$\_$reflection$\_$about$\_$uniform$\_$state$($qc$,$ inputs$)$

qc$.$barrier$()$

def create$\_$quantum$\_$circuit$\_$for$\_$grover$($n$\_$iters$)$:

inputs $=$ QuantumRegister$(4,\text{'}$b$\text{'})$

cbit $=$ ClassicalRegister$(4,\text{'}$z$\text{'})$

qc $=$ QuantumCircuit$($inputs$,$ cbit$)$

for i in inputs:

qc$.$h$($i$)$ # apply hadamard

qc$.$barrier$()$

for i in range$($n$\_$iters$)$:

Grover$\_$diffuse$($qc$,$ inputs$)$

qc$.$measure$($inputs$,$ cbit$)$

return qc

qc$2=$ create$\_$quantum$\_$circuit$\_$for$\_$grover$(4)$

print$(\text{'}$Grover Search After Four Iterations'$)$

display$($qc$2.$draw$(\text{'}$mpl$\text{'},$ style$=$'iqp'$))$

from qiskit.tools.visualization import plot$\_$histogram

# lets test after one iteration

simulator $=$ Aer.get$\_$backend$(\text{'}$aer$\_$simulator'$)$

circ $=$ transpile$($qc$2,$ simulator$)$

# Run and get counts

result $=$ simulator.run$($qc$2).$result$()$

counts $=$ result.get$\_$counts$($circ$)$

display$($plot$\_$histogram$($counts$,$ title$=$'result counts $(1024$ simulations$)\text{'}))$

assert counts$[\text{'}0011\text{'}]+$ counts$[\text{'}1010\text{'}]+$ counts$[\text{'}1100\text{'}]>=500,$ "The marked states should account for more than $50\%$ of the counts. Are you implementing the Marking circuit correctly?"$]$