Smaller state space. Hardcoded dimensions