updated prototype to have screen functionality