change Null -> Unit; add ImageAutoEncoder