Added states for gameplay