added in a couple of screen updates