change screen.c
it gives result, but it's wrong