Added Python code
Updated readme