improve image architecture