Use AngularJS binding on display's class instead of jQuery.