Added edit functionality for Issue #1.