Adding class names to the functions still some errors