Updated code
Updated readme