Mica modificare, am separat clasa de programul principal punand-o intr-un fisier separat ;)