Added State & Action (same as in the first 2 parts of the project)