Added some screen state tests