Add basic editing support