Added some code
Updated readme