Update screen.c: improve printing function
Add basic keyboard handler