update README and clean code