Added edit button for commentss and messages