add wrapper class
clean up whitespace