Updated edit button to show for author only