Improved the script and added the readme