Added missing class header, used HEREDOC instead of echo