restyle edit button and comment fields