Fixing styling issues