Imporve the script

Fix the y/n checks not working properly; Still some bugs but we'll sort them!