Little change function main

Zmiana funkcji nazw