Cleaning the code and adding README file