Fixing styling bugs