code change and readme update