add callback for focus