updating code + readme