added code in screen