Use jQuery to check focus state