Fixing styling more