Add a render function to mudle