Add conditional partial rendering