- improve the script