Code cleanup
updated readme