some tiny code changes and added results and figures