updated code/readme