egy kis Main refact