Updated

Updated the code and added readme