Prettier way of adding class