Cleaning code and readme file updated