Updated readme to suit code changes