Cleaned Code. Updated README.