Update prototype code