Added a header and added a class to main