Refactros the unicorn scritp