Switch to js selectors and fix css.