modify the main class--> three to one