separate the class and the main script