Add a state handler, woo