Changed/added classes for version stage

Changed/added classes for version stage, including the main class and a
new class