Updated readme for code