Adding the code and improving readme