Add methods for starting and stopping states