little changes to draw screen to get it to work