Updated readme

Added documentation for the code