Add some code + reformat readme.md