Fixing minor bug in python code, adding some readme code