Add trait OnFocus