extracted screen class to its own file