Added controller functionality for comment