small fixes in the script