Add optional state block