CSS fixes
Small bug correction