Added screen state tests