Adding class to solution