Changed the header to work better