Added code to set the focus.