updated code and ReadMe file