editting add comment functions to return on completion