Add headers and wheel class