Add missing import.
Add support for specifying input format from CLI