edit funtionality to comments