Update screen.cpp