moved class to header