Commit #2: Added edit function