Fixing minor visual issues