trying to enhance my script