Решил хоть с чего-то начать. Подправил main ))