Add behaviour for catch/lose focus on input