Improved the README and cleaned the code