change to use ubuntu as base image