Modify to heredoc