Implemented Edit button and removed unnecessary imports