Unnecessary import, simplify main() code.