Feature/add edit mode to directive