Add init func
Imporve the render func