code cleanup
updated the readme