Edit README and add basic code