Add edit mode for create duntionality