Updating code to map to inputs for editing.