Added edit functionality for comments