Updated Class Header File