upd code
upd func
fix code