Adding draw method to state.