added edit button/functionality