Added README file, cleaned the code