Fix import error, make program do purpose