added output class. does not work yet