Added class to version

fixes #203

May need some tweaking style wise.