keep everything identical to the original image architecture