Adding visual to if