added screen functionality