update readme

cleaned up some code & updated the readme