UI script added to code