Add ability to specify output format