Adding a class to clear the flexbox