small fix in screen two