Add ability to customize formats

Signed-off-by: Jason Chen <jhchen7@gmail.com>