do not change anything from the original image architecture