making output class an option