custom plotting function + custom x0 in mnist example