Add some basic support for 'formatter_class' argument