Better way to get current div