Add screen class which handles graphics and input