Add directive to auto focus