updated rendering to include stages