two tone pudding

I'm wondering if the pudding would come out ok if I added chocolate after cooking. In this way, I can divide pudding in half; dark chocolate for bottom, white on the top.