Fixed the class to have a header and to go along with the header