Cleaning code
Updating readme