Fix JS/CSS bugs