Add kafka code