małe poprawki w main