Added subroutine to erase screen