States (lesson 3), challenges 1 and 2 done so far