Added code
Updated readme