Modifying the render function