take base_image parameter