Added first version of the main class