Updated header guard to reflect class name