State setter. WIP