Add functionalities for getting actions in states