Header comments in class updated