np not imported, zeros_like is

Small bug in example. Numpy was not imported as np, zeros_like is.