Quick JS and CSS fix, really needs better solution