Adding a CSS code that only work on edit mode