updated readme
added docstrings and print statements