Adding more complex state example