Writer class módosítása + Main-ben megírtam a kimeneti függvényt