Add method to get current state