Fixing some visuals