Updated controller for comments