Zustand-code implementation