updating code / readme