workiing on the main class