Improved script and added README