Render callback

Add callback to render function