Add update_attributes! method which immediately applies the state