Small style update + JS fix