added headers to classes to where it was missing