[Minor] CSS and code fix