Added version class parameter

(Accidentally closed the old PR)

Added version class parameter