Adding class and header